PyTorch torch.zeros_like 関数
Pytorch torch リファレンスマニュアルtorch.zeros_likePyTorchで、入力テンソルと同じ形状の全ゼロテンソルを作成する関数です。
関数定義
torch.zeros_like(input, dtype=None, device=None, requires_grad=False)
使用例
実例
import torch
x = torch.randn(3, 4)
y = torch.zeros_like(x)
print("x 形状:", x.shape)
print("y 形状:", y.shape)
print(y)
x = torch.randn(3, 4)
y = torch.zeros_like(x)
print("x 形状:", x.shape)
print("y 形状:", y.shape)
print(y)
その他の拡張