import torch


def get_torch_zeros(shape, device='cuda'):
    """
    Generate zero tensor with given shape
    :param shape: array shape
    :param device: sets training device
    :return: zero tensor with given shape
    """
    return torch.zeros(shape, dtype=torch.float32).to(device)