Pytorch

torch.unsqueeze — PyTorch 1.13 documentation

torch.unsqueeze torch.unsqueeze(input, dim) → Tensor Returns a new tensor with a dimension of size one inserted at the specified position. The returned tensor shares the same underlying data...

torch.Tensor.expand — PyTorch 1.13 documentation

torch.Tensor.expand Tensor.expand(*sizes) → Tensor Returns a new view of the self tensor with singleton dimensions expanded to a larger size. Passing -1 as the size for a dimension means...

torch.Tensor.expand — PyTorch 1.13 documentation

torch.Tensor.scatter_add_ — PyTorch 1.13 documentation

torch.Tensor.scatter_add_ Tensor.scatter_add_(dim, index, src) → Tensor Adds all values from the tensor src into self at the indices specified in the index tensor in a similar fashion as...

torch.Tensor.scatter_add_ — PyTorch 1.13 documentation

torch.squeeze — PyTorch 1.13 documentation

torch.squeeze torch.squeeze(input, dim=None) → Tensor Returns a tensor with all the dimensions of input of size 1 removed. For example, if input is of shape: (A×1×B×C×1×D)(A \times 1...

torch.nn.functional.pad — PyTorch 1.13 documentation

torch.nn.functional.pad torch.nn.functional.pad(input, pad, mode='constant', value=None) → Tensor Pads tensor. Padding size:The padding size by which to pad some dimensions of input are described...

torch.sum — PyTorch 1.13 documentation

torch.sum torch.sum(input, *, dtype=None) → Tensor Returns the sum of all elements in the input tensor. Parameters: input (Tensor) – the input tensor. Keyword Arguments: dtype (torch.dtype,...

torch.matmul — PyTorch 1.13 documentation

torch.matmul torch.matmul(input, other, *, out=None) → Tensor Matrix product of two tensors. The behavior depends on the dimensionality of the tensors as follows: If both tensors are 1-dimensiona...

torch.transpose — PyTorch 1.13 documentation

torch.transpose torch.transpose(input, dim0, dim1) → Tensor Returns a tensor that is a transposed version of input. The given dimensions dim0 and dim1 are swapped. If input is a strided tensor...

torch.add — PyTorch 1.13 documentation

torch.add torch.add(input, other, *, alpha=1, out=None) → Tensor Adds other, scaled by alpha, to input. outi=inputi+alpha×otheri\text{{out}}_i = \text{{input}}_i + \text{{alpha}} \times...

torch.add — PyTorch 1.13 documentation

torch.Tensor.view — PyTorch 1.13 documentation

torch.Tensor.view Tensor.view(*shape) → Tensor Returns a new tensor with the same data as the self tensor but of a different shape. The returned tensor shares the same data and must have...